#include <iostream>
#include <string>
#include <vector>
#include <pthread.h>
#include <unistd.h>
#include <functional>
#include "Thread.hpp"
#include "BlockQueue.hpp"

using namespace ThreadModule;

int a = 10;

void Consumer(BlockQueue<int> *&bq)
{
    while (true)
    {
        int data;
        bq->Pop(&data);
        std::cout << "Consumer consum data is : " << data << std::endl;
        // sleep(1);
    }
}
void Productor(BlockQueue<int> *&bq)
{
    while (true)
    {
        int cnt = 1;
        while (true)
        {
            bq->Enqueue(cnt);
            std::cout << "Productor product data is: " << cnt << std::endl;
            cnt++;
            sleep(1);
        }
    }
}
/*  --------- 测试bq是不是用的同一个 -----------   */
// void Product(BlockQueue* &bq)
// {
//     while(true)
//     {
//         int cnt;
//         std::cout << "Product : " << bq << std::endl;
//         sleep(1);
//     }
// }
// void Consum(BlockQueue* &bq)
// {
//     while(true)
//     {
//         int data;
//         std::cout << "Consum : " << bq << std::endl;
//         sleep(1);
//     }
// }

void StartComm(std::vector<Thread<BlockQueue<int> *>> *threads, int num, BlockQueue<int> *bq, func_t<BlockQueue<int> *> func)
{
    for (int i = 0; i < num; i++)
    {
        std::string name = "thread-" + std::to_string(i + 1);
        threads->emplace_back(func, bq, name);
        threads->back().Start();
    }
}
void StartConsumer(std::vector<Thread<BlockQueue<int> *>> *threads, int num, BlockQueue<int> *bq)
{
    StartComm(threads, num, bq, Consumer);
}
void StartProductor(std::vector<Thread<BlockQueue<int> *>> *threads, int num, BlockQueue<int> *bq)
{
    StartComm(threads, num, bq, Productor);
}

void WaitAllThread(std::vector<Thread<BlockQueue<int> *>> &threads)
{
    for (auto &thread : threads)
    {
        thread.Join();
    }
}

int main()
{
    // bq是一个指针，指针传参时会发生值拷贝
    BlockQueue<int> *bq = new BlockQueue<int>(5);
    std::vector<Thread<BlockQueue<int> *>> threads;

    // bq是一个指针，指针传参时会发生值拷贝，所以这里传过去的是一个临时的指针对象，
    // 即传给线程的数据是一个临时的指针对象，而线程里的数据又是引用，引用一个临时对象，
    // 这个临时对象可能已经释放了。传给Consumer和Productor时这个临时指针可能已经被释放了，
    // 所以在向阻塞队列push和pop时，有可能他对应的阻塞队列的对象已经不存在了，
    // 对象不存在了，我们再去push就会发生段错误，就发生了野指针问题
    StartConsumer(&threads, 1, bq);
    StartProductor(&threads, 1, bq);
    WaitAllThread(threads);
    return 0;
}

/*-------------------------  初步搭成 生产消费模型  ------------------------------------*/
// #include <iostream>
// #include <vector>
// #include <pthread.h>
// #include <unistd.h>
// #include "Thread.hpp"

// using namespace ThreadModule;

// int a =10;
// void Product(int &cnt)
// {
//     while(cnt)
//     {
//         std::cout << "Product : " << cnt-- << std::endl;
//         sleep(1);
//     }
// }
// void Consum(int &cnt)
// {
//     while(cnt)
//     {
//         std::cout << "Consum : " << cnt-- << std::endl;
//         sleep(1);
//     }
// }

// void StartProduct(std::vector<Thread<int>> *threads, int num)
// {
//     // 创建生产者线程
//     for (int i = 0; i < num; i++)
//     {
//         std::string name = "thread-" + std::to_string(i+1);
//         threads->emplace_back(Product, a, name);
//         // 启动线程
//         threads->back().Start();
//     }
// }
// void StartConsume(std::vector<Thread<int>> *threads, int num)
// {
//     // 创建消费者线程
//     for (int i = 0; i < num; i++)
//     {
//         std::string name = "thread-" + std::to_string(i+1);
//         threads->emplace_back(Consum, a, name);
//         // 启动线程
//         threads->back().Start();
//     }
// }
// void WaitAllThread(std::vector<Thread<int>> &threads)
// {
//     for(auto & thread : threads)
//     {
//         thread.Join();
//     }
// }

// int main()
// {
//     std::vector<Thread<int>> threads;

//     StartProduct(&threads, 1);
//     StartConsume(&threads, 1);
//     WaitAllThread(threads);
//     return 0;
// }